'''
生成随机数，纯数字，传参为random*
*为生成的长度
'''

import random
import string

class Merhod(object):
    def generate_random_string(self, input_str):
        # 获取随机字符串的长度
        length = self.get_length_from_str(input_str)
        # 生成随机字符串
        letters_and_digits = string.ascii_letters + string.digits
        return ''.join(random.choice(letters_and_digits) for _ in range(length))

    def get_length_from_str(self, input_str):
        # 找到数字部分的起始位置
        i = len(input_str)
        while i > 0 and input_str[i - 1].isdigit():
            i -= 1
        # 返回数字部分的整数值作为长度
        return int(input_str[i:]) if i < len(input_str) else 0

if __name__ == '__main__':
    # 实例化 Merhod 类并生成随机字符串
    merhod = Merhod()
    result = merhod.generate_random_string('random5')
    print(result)
